package com.sheng.leetcode.year2023.month06.day05;

import org.junit.Test;

import java.util.Arrays;

/**
 * @author liusheng
 * @date 2023/06/05
 * <p>
 * 2460. 对数组执行操作<p>
 * <p>
 * 给你一个下标从 0 开始的数组 nums ，数组大小为 n ，且由 非负 整数组成。<p>
 * 你需要对数组执行 n - 1 步操作，其中第 i 步操作（从 0 开始计数）要求对 nums 中第 i 个元素执行下述指令：<p>
 * 如果 nums[i] == nums[i + 1] ，则 nums[i] 的值变成原来的 2 倍，nums[i + 1] 的值变成 0 。否则，跳过这步操作。<p>
 * 在执行完 全部 操作后，将所有 0 移动 到数组的 末尾 。<p>
 * 例如，数组 [1,0,2,0,0,1] 将所有 0 移动到末尾后变为 [1,2,1,0,0,0] 。<p>
 * 返回结果数组。<p>
 * 注意 操作应当 依次有序 执行，而不是一次性全部执行。<p>
 * <p>
 * 示例 1：<p>
 * 输入：nums = [1,2,2,1,1,0]<p>
 * 输出：[1,4,2,0,0,0]<p>
 * 解释：执行以下操作：<p>
 * - i = 0: nums[0] 和 nums[1] 不相等，跳过这步操作。<p>
 * - i = 1: nums[1] 和 nums[2] 相等，nums[1] 的值变成原来的 2 倍，nums[2] 的值变成 0 。数组变成 [1,4,0,1,1,0] 。<p>
 * - i = 2: nums[2] 和 nums[3] 不相等，所以跳过这步操作。<p>
 * - i = 3: nums[3] 和 nums[4] 相等，nums[3] 的值变成原来的 2 倍，nums[4] 的值变成 0 。数组变成 [1,4,0,2,0,0] 。<p>
 * - i = 4: nums[4] 和 nums[5] 相等，nums[4] 的值变成原来的 2 倍，nums[5] 的值变成 0 。数组变成 [1,4,0,2,0,0] 。<p>
 * 执行完所有操作后，将 0 全部移动到数组末尾，得到结果数组 [1,4,2,0,0,0] 。<p>
 * <p>
 * 示例 2：<p>
 * 输入：nums = [0,1]<p>
 * 输出：[1,0]<p>
 * 解释：无法执行任何操作，只需要将 0 移动到末尾。<p>
 * <p>
 * 提示：<p>
 * 2 <= nums.length <= 2000<p>
 * 0 <= nums[i] <= 1000<p>
 * <p>
 * 来源：力扣（LeetCode）<p>
 * 链接：<a href="https://leetcode.cn/problems/apply-operations-to-an-array">2460. 对数组执行操作</a><p>
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。<p>
 */
public class LeetCode2460 {

    @Test
    public void test01() {
//        int[] nums = {1, 2, 2, 1, 1, 0};
        int[] nums = {0, 1};
        System.out.println(Arrays.toString(new Solution().applyOperations(nums)));
    }
}

class Solution {
    public int[] applyOperations(int[] nums) {
        int n = nums.length;
        for (int i = 0; i < n - 1; i++) {
            if (nums[i] == nums[i + 1]) {
                nums[i] = 2 * nums[i];
                nums[i + 1] = 0;
            }
        }
        int index = 0;
        int[] ans = new int[n];
        Arrays.fill(ans, 0);
        for (int num : nums) {
            if (num != 0) {
                ans[index++] = num;
            }
        }
        return ans;
    }
}
